Robot assisted cognitive training